package cn.jietuo.juc.c_007_reentrant;

import java.util.Random;
import java.util.concurrent.Phaser;
import java.util.concurrent.TimeUnit;

/**
 * @author zhangx & jietuo_zx@163.com
 * @version 1.0
 * @date 2020/5/7
 * @description: Phaser的使用
 * 例如有这样的一个题目：
 * 5个学生一起参加考试，一共有三道题，要求所有学生到齐才能开始考试，
 * 全部同学都做完第一题，学生才能继续做第二题，
 * 全部学生做完了第二题，才能做第三题，
 * 所有学生都做完的第三题，考试才结束。
 * 分析这个题目：这是一个多线程（5个学生）分阶段问题
 * （考试考试、第一题做完、第二题做完、第三题做完），
 * 所以很适合用Phaser解决这个问题。
 * <p>
 * phase表示阶段，party表示每个阶段的线程个数，
 * 只有每个线程都执行了phaser.arriveAndAwaitAdvance();
 * 才会进入下一个阶段，否则阻塞等待。
 */
public class T06_Phaser {


    static class MyPhaser extends Phaser {

        @Override
        protected boolean onAdvance(int phase, int registeredParties) {
            switch (phase) {
                case 0:
                    return studentArrived();
                case 1:
                    return finishFirstExercise();
//                case 2:
//                    return finishSecondExercise();
                case 2:
                    return finishExercise();
                default:
                    return true;
            }
        }

        private boolean finishExercise() {
            System.out.println("第二题完成，完成考试");
            return true;
        }

        private boolean finishFirstExercise() {
            System.out.println("完成第一题");
            return false;
        }

        private boolean studentArrived() {
            System.out.println("学生准备好了,学生人数：" + getRegisteredParties());
            return false;
        }
    }

    static class StudentTask implements Runnable {

        private MyPhaser myPhaser;

        public StudentTask(MyPhaser myPhaser) {
            this.myPhaser = myPhaser;
        }

        @Override
        public void run() {
            System.out.println(Thread.currentThread().getName() + "---到达参加考试");
            myPhaser.arriveAndAwaitAdvance();

            System.out.println(Thread.currentThread().getName() + "---开始做第一题");
            doExercise1();
            System.out.println(Thread.currentThread().getName() + "---第一题做完");
            myPhaser.arriveAndAwaitAdvance();


            System.out.println(Thread.currentThread().getName() + "---开始做第二题");
            doExercise2();
            System.out.println(Thread.currentThread().getName() + "---第二题做完");
            myPhaser.arriveAndAwaitAdvance();

        }

        private void doExercise1() {
            Random random = new Random(10);
            try {
                TimeUnit.SECONDS.sleep(random.nextInt());
            } catch (InterruptedException e) {
                e.printStackTrace();
            }

        }

        private void doExercise2() {
            Random random = new Random(10);
            try {
                TimeUnit.SECONDS.sleep(random.nextInt());
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    public static void main(String[] args) {
        MyPhaser myPhaser = new MyPhaser();
        Thread[] threads = new Thread[5];
        for (int i = 0; i < threads.length; i++) {
            //注册一次表示phaser维护的线程个数
            myPhaser.register();
            threads[i] = new Thread(new StudentTask(myPhaser),"student" +i);
        }
        for (int i = 0; i < threads.length; i++) {
            threads[i].start();
        }

        for (int i = 0; i < threads.length; i++) {
            try {
                threads[i].join();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        System.out.println("pharse has finished  " + myPhaser.isTerminated());
    }

}
